# -*- coding: utf-8 -*-
'Extract BusLineNames after POI-Geting'
'''Create By Sidchen in 2020'''

import xlrd,xlwt
import os

print('Extract BusLineNames from POI data (bus station)')
if not os.path.exists('Linedata/POI.xls'):
    input('找不到Linedata/POI.xls,请确认')
    exit()
else:
    print('正在打开POI.xls文件读取数据…………')
#打开POI.xls
poi=xlrd.open_workbook('Linedata/POI.xls')
table=poi.sheets()[0]
#遍历所有行，提取地址列（column=4，row 1~n）
Busline=set()
for i in range(table.nrows-1):
    k=table.cell_value(i+1,4)
    #每个地址单元格依据'；'符号提取单个线路名称，存入set（），即使线路名称有（说明）也无所谓
    for j in k.split(';'):
        Busline.add(str(j))
if os.path.exists('Linedata/BusLineNames.txt'):
    os.remove('Linedata/BusLineNames.txt')
name=open('Linedata/BusLineNames.txt','a',encoding='utf-8')
# 提取城市
city = table.cell_value(1, 11)
name.write(city+'\n')
#将set（）写入BusLines.txt中
for i in Busline:
    name.write(i+'\n')
name.close()
input('公交线路名称已提取完成，共获得'+str(len(Busline))+'条公交线路,任意键退出')
exit()
